#! /bin/bash

atc --framework=5 --model=./ReID_dybs.onnx --input_format=NCHW --input_shape="image:1,3,256,128" --output=ReID_aipp_bs1 --log=debug --soc_version=Ascend310 --insert_op_conf=./aipp_ReID_pth.config

ln -s /root/datasets/market1501 ./market1501
python3.7 gen_dataset_info.py jpg ./market1501/query ./market1501_query.info
python3.7 gen_dataset_info.py jpg ./market1501/bounding_box_test ./market1501_box.info

./benchmark.x86_64 -model_type=vision -device_id=0 -batch_size=1 -om_path=./ReID_aipp_bs1.om -input_text_path=./market1501_query.info -input_width=128 -input_height=256 -output_binary=True -useDvpp=True

./benchmark.x86_64 -model_type=vision -device_id=0 -batch_size=1 -om_path=./ReID_aipp_bs1.om -input_text_path=./market1501_box.info -input_width=128 -input_height=256 -output_binary=True -useDvpp=True
python3.7 ReID_postprocess.py --query_dir=./market1501/query --gallery_dir=./market1501/bounding_box_test --pred_dir=./result/dumpOutput_device0 > ./result_aipp_bs1.json
